#!/usr/bin/env python
# coding:utf-8

import requests
from hashlib import md5


# TODO  错误代码对照表 https://www.chaojiying.com/api-23.html
class Chaojiying_Client(object):

    def __init__(self, username, password, soft_id):
        self.username = username
        password =  password.encode('utf8')
        self.password = md5(password).hexdigest()
        self.soft_id = soft_id
        self.base_params = {
            'user': self.username,
            'pass2': self.password,
            'softid': self.soft_id,
        }
        self.headers = {
            'Connection': 'Keep-Alive',
            'User-Agent': 'Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 5.1; Trident/4.0)',
        }

    def PostPic(self, im, codetype):
        """
        im: 图片字节
        codetype: 题目类型 参考 http://www.chaojiying.com/price.html
        """
        params = {
            'codetype': codetype,
        }
        params.update(self.base_params)
        files = {'userfile': ('ccc.jpg', im)}
        r = requests.post('http://upload.chaojiying.net/Upload/Processing.php', data=params, files=files, headers=self.headers)
        return r.json()

    def PostPicPath(self, img_path, codetype=1902):

        params = {
            "codetype": codetype,
        }
        params.update(self.base_params)
        i = b""
        with open(img_path, "rb") as f:
            i = f.read()
        files = {"userfile": ("ccc.jpg", i)}
        r = None
        while True:
            try:
                r = requests.post("http://upload.chaojiying.net/Upload/Processing.php", data=params, files=files,
                                  headers=self.headers)
            except Exception as err:
                pass
            else:
                if r.status_code == 200:
                    break
        return r.json()

    def ReportError(self, im_id):
        """
        im_id:报错题目的图片ID
        """
        params = {
            'id': im_id,
        }
        params.update(self.base_params)
        r = None
        while True:
            try:
                r = requests.post('http://upload.chaojiying.net/Upload/ReportError.php', data=params, headers=self.headers)
            except Exception as err:
                pass
            else:
                if r.status_code == 200:
                    break
        return r.json()


if __name__ == '__main__':
    chaojiying = Chaojiying_Client('17505251998', 'yinbohang1', '905824')
    url = 'https://tyrz.chinatorch.org.cn/hjismp/servlet/validateCodeServlet'
    for i in range(10):
        rst = requests.get(url=url)
        with open('img/%d.jpg'%i, 'wb') as e:
            e.write(rst.content)
        im = open('img/%d.jpg'%i, 'rb').read()
        print(chaojiying.PostPic(im, 1902))

